#include <memory>
#include "UdpServer.hpp"    // 网络通信服务器模块
#include "Route.hpp"        // 路由模块
#include "ThreadPool.hpp"   // 线程池模块
#include "Log.hpp"

using namespace ThreadPoolModule;

// 回显服务经常用于检测
std::string defaultHandler(const std::string &message)
{
    std::string s = "server say@ ";
    s += message;
    return s;
}

using task_t = std::function<void()>;

// 通过命令行 ./udpserver port 启动服务器
int main(int argc, char *argv[])
{
    if (argc != 2)
    {
        std::cerr << "Usage: " << argv[0] << " port" << std::endl;
        return 1;
    }
    uint16_t port = std::stoi(argv[1]);
    Enable_Console_Log_Strategy();

    // 1. 路由服务
    Route r;

    // 2. 线程池单例
    auto tp = ThreadPool<task_t>::GetInstance();

    // 3. 创建网络服务器对象，提供通信功能
    std::unique_ptr<UdpServer> usvr = std::make_unique<UdpServer>(port, [&r, &tp](int sockfd, const std::string &message, InetAddr &peer){
        task_t t = std::bind(&Route::MessageRoute, &r, sockfd, message, peer);
        // // 将任务给到线程池
        tp->Enqueue(t);
    });

    // 4. 启动服务器
    usvr->Init();
    usvr->Start();
    return 0;
}